Trying to run different functions based on user input